import os


# 清空文件夹
def clear_folder(path):
    for root, dirs, files in os.walk(path):
        for f in files:
            file_path = os.path.join(root, f)
            os.remove(file_path)
        for d in dirs:
            dir_path = os.path.join(root, d)
            clear_folder(dir_path)
            os.rmdir(dir_path)


# 递归统计文件夹内文件个数
def count_files(directory):
    # 初始化文件数量计数器
    count = 0
    # 列举文件夹内的所有项目
    for item in os.listdir(directory):
        # 获取完整的路径
        item_path = os.path.join(directory, item)
        # 如果是文件，增加计数器
        if os.path.isfile(item_path):
            count += 1
        # 如果是文件夹，递归调用函数并累加结果
        elif os.path.isdir(item_path):
            count += count_files(item_path)
    # 返回文件数量总计
    return count


def get_model_path(root):
    return os.path.join(root, "model")


def get_model_info_path(root):
    return os.path.join(root, "info")


def get_model_log_path(root):
    return os.path.join(root, "log.txt")